package com.itany.corejava.code13_jdbc;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author 石小俊
 * @date 2023年05月29日 10:23
 */
public class Test07_保存返回主键 {

    public static void main(String[] args) {
        User user = new User("李白", "123456", "13912367896", "王者荣耀");
        try {
            Integer id = insertReturnPrimaryKey(user);
            System.out.println("保存后返回的主键值为:" + id);
        } catch (DataAccessException e) {
            e.printStackTrace();
        }

    }

    public static Integer insertReturnPrimaryKey(User user) throws DataAccessException {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Integer id = null;
        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append(" insert into ")
                    .append(" t_user ")
                    .append("   (username,password,phone,address)")
                    .append(" values ")
                    .append("   (?,?,?,?)")
                    .toString();
            // 想要实现保存返回主键的功能
            // 在获取状态集的时候,需要设置状态集操作时保存返回主键
            // 存在第二个参数,用于设置保存是否返回自动生成的键的标志
            // 其值是由Statement提供的两个常量
            // RETURN_GENERATED_KEYS: 返回自动生成的键
            // NO_GENERATED_KEYS: 不返回自动生成的键
            // PreparedStatement是Statement的子类,可以继承对应的常量
            ps = conn.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
            ps.setString(1, user.getUsername());
            ps.setString(2, user.getPassword());
            ps.setString(3, user.getPhone());
            ps.setString(4, user.getAddress());
            // 执行DML操作
            ps.executeUpdate();
            // 当执行DML操作之后,在数据库中会生成对应的主键的值
            // 此时我们需要获取到生成后的主键的值
            // Statement对象提供了获取自动生成的键的方法
            // 其返回值也是一个结果集,但是我们对应其返回的键的列名可能并不知道
            rs = ps.getGeneratedKeys();
            while (rs.next()) {
                // 由于生成的主键是由数据库自动操作的
                // 对于开发者而言,数据库厂商为其定义的列名我们可能并不知道
                // 那么此时如何映射?
                // 此时的开发者虽然不知道列名,但是能够确定的是返回的值有且仅有一列
                // 因此,可以通过索引号进行映射,直接将第一列的数据映射给对应的主键值
                id = rs.getInt(1);
            }
            return id;
        } catch (DataAccessException e) {
            throw e;
        } catch (SQLException e) {
            throw new DataAccessException("数据访问异常");
        } finally {
            JDBCUtil.close(conn, ps, rs);
        }
    }

}
